'''
* This is the projet for Brtc LlmOps Platform
* @Author Leon-liao <liaosiliang@alltman.com>
* @Description //TODO 
* @File: 3_study_bat_handle.py
* @Time: 2025/7/10
* @All Rights Reserve By Brtc
'''
from datetime import datetime

import dotenv
from langchain_community.chat_models import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate

dotenv.load_dotenv()

# 1、编排 prompt
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是OpenAi 开发的机器人， 请回答用户的问题， 现在的时间是 {now}"),
    ("human","{query}")
]).partial(now = datetime.now())

#2、创建大语言模型
llm = ChatOpenAI(model = "gpt-3.5-turbo-16k")


ai_message = llm.batch([
    prompt.invoke({"query":"现在是几点钟了， 30字"}),
    prompt.invoke({"query":"现在是几点钟了， 30字"}),
    prompt.invoke({"query":"现在是几点钟了， 30字"}),
    prompt.invoke({"query":"现在是几点钟了， 30字"})
])


print(len(ai_message))
print(type(ai_message))
for one_message in ai_message:
    print("=================================")
    print(one_message.content)